Add code checks