Add getInitialOnes method