Reorganised models folder
Reworked paths for notebooks
Started work on class-based implementation
Updated README