Mihir , Vikash | Removing needless flatten