Consolidating pygame code into a single class.