Added better legend