Reorganized file structure

Now there is a models folder