Version 2.1
Used simpler code in the end.
modified:   03.primAssignment.py