Added return object for function