Update add function too allow for custom icon