modified code and adding problems folder