added icon option
fixed styling