fixed some edge bugs