Fix issues with tree writer for modules and default export

Fixes #439

Review URL: https://github.com/google/traceur-compiler/pull/442

Closes #442, closes #439.