Python/PyGame TD game