Changed return value of `draw` and `input`

This makes it easier to preserve return values