class issue fixed for treemap, colors added, pushing assignment 4