Tested version using random forrest algorith