Fixes #36. Add missing use statements