fixed end set layout to be pretty