Generate a main.py python script that will return the maximum of two numbers.
a = 3;
b = 4
def max(a, b): 
    if a > b: 
        return a 
    else:  
        return b 
  
#Driver Code 
a = 3
b = 4
print(max(a, b))