Add classes to test gamestates