Update Functions.R

Fixed bug on `check_transition_probability`. It worked on 3D arrays but not on 2D matrices. Ive implemented the fix by checking it's a 2D and if it is, will make it into a 3D array with only one slice on the 3rd dimension

Author: feralaes