added more code because i got my computer working