Bug fixes + icons