Adding some functionality to build script.