Simple function override in python