Updated getOnes function