Updating with return_type

Added return_type