more code correction and added Div class