Using PyGame lib to write a simple game and do some improvement