modyfying python code as modules