Update A.scala

Correct sol