More annotations. Code is now working. Requires matrix input.
Some more annotation required.