Write unit tests for each function

[cameronove]

Each function needs to have a unit test specific to that funciton