add equation to main