Modified to print expected answer