Implemented suggestion #64

Signed-off-by: Ken 'Minalien' Murray <minalien@gmail.com>
Merge pull request #104 from Minalien/master

Implemented suggestion #64