(Done) changed the return type