Add response code logging