fixed elongation bug. reshape problem.  using squeeze instead.