Moved rendering code to separate class
Moved saving and loading code to separate class