Made small change with returning values in library