add module prediction