Making some changes to the build method