Small updates to build file