start to implement `while` - not currently working