updated the layout slightly