Added the mainfile still need to split out the classes etc.