updated the layout a litlle