Added the new code with css changes