Fix for issue #9. Added required import statement.