Update stage_two.py

Stage 2 for CODE2040 Application