WIP. Added docs and updating readme