incorperated in main code

Fixed display:none bug