If-Else working. Added square gallery code.