Solution for HW1 - CS255