Edited edit function xD