add small icon logic right