Toy pygame experiment