added legend
minor fix
styling