Add hot code reload