Added the pygame controller class