fixing layout a little