Added tests.py + implemented FP detection with mask