add icon code
cleanup