wrote some code
screenshot