- Removing logging statements
- Fixed failed test case