revert css and added some more code