More work on the level generation