Working before attempting to implement viscosity
Added viscosity
Fixed bugs