Fixed mean_pool code and added sum_pool in mlp