Added function collatz_step and unit tests