Added return type to doc block