Put this below the scala import