Bug fixed - return code. Result is printed correctly