Optimize with reshaper