Added variable assignment and while loop