Minor layout fixed