hwk4 part1 - Added layout