Add method to get the current step number