hw0 - part1 working