Switched code to easier MIT license