CSC148 Assignment 2 : A tree map program in python