Modified CSS to work