Issue #3 solution (aparently)

I tried to recreate the issue #3 problem and it seemed that the code was working with these little fix, also, the plugin is in trending