Add Python version of homework