Created classes and main for initial version