Move everything in an init file as currently its just a simple script