Add function fix attempt 1