Refine flex responsive layout