Updated to remove stupid bug.