Add custom move logic