started adding some tests. better late than never