add output flag, minor bug fixes