Add css styles and small change in js