Writing custom layouts for better UX