layout now working with css flex