adding few lines of code