First commit to solution.py