Optimal design for multi-arm multi-stage clinical trials