Updated Edit.java

Added a few if statements