adding a few lines of code