Added missing css classes to make everything work