Writing Function In Python