Add debug logic