working code

This is the first version of a working code that produces desired output from given inout. Still needs te be checked for physics accuracy.