Adding more css and functionality