cleaned up repo, added small modifications