Small pygame experiement