Extension to Latex which makes assignment write-ups easier to write