Refactor live reload module