add floor-puzzle.py