Could we use Lombok here?