Update Flatten, Readme

Added a flag argument to flatten to control the types to flatten.
Updated README to reflect this change