Adding correct display/hide