ex4.py reworked

Added comments and study drills