Removed logs, cleaned code, added a function param

I'm back to work on this :) Soon 1.1 version