Started work on HW2

Finished the simple part of the HW, still needs optimizations