cleaned up edit layout