cleaned up and documented code + changed one picture