Quick Additions

I added a few lines of code to make things a little easier in the
future.