Added for and while statements