Added 4 to potential function