Adding homework 2 part 3
added 4x4 option