Cleaned up the code after submitting FJOS paper.