added comments to prediction.py