Update main.f90

3d now