Grid layout is now looking properly