Fix module importing
Add images
Updated readme