added legend and made visual attributes better