Small fixes

Removed obsolte tranfer between host & device
Fixed expression for calculating the time step size