Add layout. Fix styles