Some alteration in code