Added action prediction