Fixing code structure with kasomas code