changed the layout slightly