edited tests and added a few more comments