[Klax] ensure the environment scales correctly in difficulty

[PaulSeitz]

Compare the later difficulties of ALE Klax and JAXtari Klax and make sure JAXtari Klax has all the later modes and options