Reworked module for better setup. Updated readme