small changes to flattener app