Initial work towards implementing the pulse propagation code