ENH: Refactor and clean user interface
ENH: Add missing test labelmap and update test script