Media query layout improvements