CS201 Assignment. 2nd assignment. Focus on arrays