Minor UI fixes

very little changes, just fixed container class issues.