Added some additional functionality