Tidy up init code