Changed the element hiding setup