. small changes + adding unit testing