Pulled code check lines

Pulled out the check lines to ensure it was wokring