added unit tests prototype