Added advanced prediction model to Future work