adding forecast functionality