Added code made before repo. Last commit was wrong