Added method to check for check
Fixed pre-exisiting bug in tests