Quick code update
I'm going to bed. Working on implementing print output