Add a step for the end of the function for functions that do not have return values

[ferranbt]

If a function has a return value, there is a step at the end of the function for that statement, but, if the function does not have any return statements, the last statement is the last one.