Add Justyna modifications

Closes #5
Added variable cojes which does nothing