Added a class to hide things properly