Edited generator to add the build task