Trying to design main classes;