Improve maze example