Updated test and readme