Added prediction viz and code.