Adding minor fixing to layout