Setup zaxis option
Fixed legend conditions (zaxis option)