Program that creates a Simple fucntion on python