fixed layout a bit