Created models package. Organised code better.