Add a separate .py file to hw