Add pygame code