mini code clean-up