Add main code to repo