Those hats feel likda necessary to the whole vibe, I would go for the second version, personally. Also, I feel like the melody is a little repetitive. Maybe change it for the second time around, or double its length? The structure works fine, though! (I think)