Added some stuff

I added this one line and think you should add it to your project.