Square and cube challenge