Modify setup file
1. Add function to set bundle