Implementing model
Continued with model

Rewrite the find_moment function to ensure that it finishes