solution.py : script 1 and 2 together