Update reader.py to focus on next step