Updated readme and added minor test