looks fine, just put ```return``` outside of if-else and then you can remove whole ```else``` part ;-)
I have no exp. with this so I'll merge in good faith, no place to test it really.