Moved getIcon to models