Fixing template

Some files were missing, and a import statement was wrong.