Added the ability to specify custom templates. Closes #2.