Worked on answer logic