a simple program of a square bouncing back and forth with some color effects in pygame