Looks fine
Some variable tweaks, and func now has return.