Added a few lines of code