Bug fix & readme