Layout fix

Fixing layout