Readd some code lost by git