add while loop, not quite tho