Update module4.py

cleaned for thesis input