FIX- Tidy code