Modifications to the matlab visualization