Creating user defined modules in Python