Added forecast functionality