Edge Case Edited

Edge case changed to better work flow of code.