How can this simulation be made realistic? Here is a short description and code.