added discussion and future work and a few tweaks