Working on predict page