Refactoring

Created Models folder for better file organization