not working but getting close
Added enum