working on logic to change icon