Lil changes in build script