Code Cleanups

Made main() less awkward for everyone involved
Made DCT() compatible with matrices other than 8x8s
DCT() now checks whether its argument matrix is square
Added docstring