Fixed layout

Fixed the layout from #1.