animation using python and pygame