Create five house problem.py