Initial commit
Updated main logic and model classes