Fixed the map a bit; updated TODO