Added missing use statement. Fixes #11.