Little improvements for assignment 6
When in a leaf, no reason to error