formatted code
use equals instead of ==