still need to add icon functionality