Finally got matrix multiply working