added legend & styling