Added a function for convenient asserting
Added initial test units