__main__ check added