add while and do loops