Fixing layout a little bit
Fixing layout a little bit