Added models folder
Changed the code
Merge pull request #1 from james-hartman/Code

Code