Trainning game  Python using Pygames