Small 1st year matlab project