Added setter functions and changed the order in which the class works.
Removed .idea directory