reworking my mod with new ideas