Moved matrix calculations to .py file
Looking at wave speed