More readability

• The main file contains the minimal details required. Many helper functions are moved into utils file.
• Model's API is simplified a lot.

---

• Speed issues remain to be solved.