Fix code
Add icon