Fix for issue 4. Can override the generated package

I needed this for a project that I'm working on. I took a simple, approach, hopefully it is suitable. I didn't write any unit tests, but did test this code against my local project. Hopefully it is suitable.