added some display none stuff