updated layout a bit