Added more to .py file