Fixed division by 0 error.
Added print statements to match return