Add a docstring for the station_prediction method.