Add screenshots & move maze generator code into better location