Added some classed to code