Some edits, removed difficulty setter for now.