edit

made better and cleaner