Added some code.

Creating pull request.