Small update

Nothing major, just added the icon, minor code changes.