Added pygame animation