Simple maze generator in Python