Added test and improved readme