Added test for set_tax
Added documentation. Merged two tests as they are entirely co-dependent.