Adding stock & fixing AI