Added some tests and fixed readme