Computational model for Go and Partial Stop trials