added and improved code

Getting better at this