Fixed issue #5

Hope this helps! :D