Added method to retrieve result based on input (int)