Completed unit tests and added comment in the README