Added support for more flags
Updated the readme