Improved hero select screen, removed unused imports.