Updates for open sourcing on GitHub