Added missing import
Fixed bug on multiple inserts
Added first unit tests