Fixed a bug and added answer files