Fix: imports after doc
- Add image class
- Add tests
Add function description
Rename test_ file name