updated and cleaned up repo