Testing pygame