Added the relevant stuff.
Merge pull request #1 from SpinyCucumber/master

Added the relevant stuff.