Runway (NYC) interview question

Complete implementation of Python code, introduce tests