Segment in 4 classes
Cleaner code