Completed the stock picker script