Slight change

bug fixes