Add prodiction for movement