Add return values to functions