Moving to .py file