Basic math module working / some unit tests