Clean up code
Fix local repo