Moved most of the graphics to a seperate class. (errors)