Paint application in python using pygame