Code tidy :)