Prof. gave some hints on some faster and cleaner code