just a lil modification